1326C - Permutation Partitions - CodeForces Solution

#include <bits/stdc++.h>
using namespace std ;

#define int long long
#define test long long t ; cin >> t ; while (t--)
#define gofast() ios_base::sync_with_stdio(false) ; cin.tie(0) ; cout.tie(0)
#define mod (int)998244353

int32_t main() {
    gofast() ;

    int n , k ;
    cin >> n >> k ;
    int a[n] ;

    for (int i=0 ; i<n ; i++)
        cin >> a[i] ;

    int ans = 1 ;
    for (int i=0 , j=-1 ; i<n ; i++) {
        if (a[i] > n-k) {
            if (j == -1)
                j = i ;

            else {
                ans *= i-j ;
                ans %= mod ;
                j = i ;
            }
        }
    }

    cout << k*(n+n-k+1)/2 << " " << ans ;

    return 0 ;
}